package com.ghy.demo3.branchwarehouse;

/**
 * 数据源切换
 */
public class DataSource {
    // 默认数据源
    public final static String DEFAULT_SOURCE = null;
    private final static ThreadLocal<String> local = new ThreadLocal<String>();
    private DataSource(){}

    /*** 清空数据源 */
    public static void clear() { local.remove(); }

    /*** 获取当前正在使用的数据源名字 */
    public static String get() { return local.get(); }

    /*** 还原当前切面的数据源 */
    public static void restore() { local.set(DEFAULT_SOURCE); }

    /*** 设置已知名字的数据源  */
    public static void set(String source) { local.set(source); }

    /*** 根据年份动态设置数据源 */
    public static void set(int year) { local.set("DB_" + year); }
}
